package com.yun.algorithmproblem.leetcode;

public class Leetcode904 {

    public static void main(String[] args) {
        Leetcode904 obj = new Leetcode904();
        int[] fruits = new int[]{1, 0, 1, 4, 1, 4, 1, 2, 3};
        int result = obj.totalFruit(fruits);
        System.out.println(result);
    }

    public int totalFruit(int[] fruits) {
        int length = fruits.length;
        if (length <= 2) {
            return length;
        }

        int prevType = -1;       // 前一种水果类型
        int currType = -1;       // 当前主要水果类型
        int currStreak = 0;      // 当前类型连续出现的次数
        int windowSize = 0;      // 当前窗口长度
        int maxWindow = 0;       // 最大窗口长度（结果）

        for (int type : fruits) {
            if (currType == -1) {
                // 初始化当前类型
                currType = type;
                currStreak = windowSize = maxWindow = 1;
            } else if (currType == type) {
                // 与当前类型相同，窗口扩展
                currStreak++;
                windowSize++;
            } else {
                if (prevType == type) {
                    // 与前一种类型相同，窗口扩展
                    windowSize++;
                } else {
                    // 遇到第三种类型，窗口收缩
                    windowSize = currStreak + 1;
                }
                // 更新类型记录
                prevType = currType;
                currType = type;
                currStreak = 1;
            }
            // 更新最大窗口长度
            maxWindow = Math.max(maxWindow, windowSize);
        }
        return maxWindow;
    }
}
